const llm_url = import.meta.env.VITE_LLM_URL as string;
const llm_key = import.meta.env.VITE_LLM_KEY as string;
const llm_model = import.meta.env.VITE_LLM_MODEL as string;

// 调用模型回复消息
export const replyMessage = async ({prompt, query}: {prompt?: string, query?: string}) => {
	return fetch(llm_url + '/stream', {
		method: 'POST',
		headers: {
			'Content-Type': 'application/json',
			'Authorization': `Bearer ${llm_key}`
		},
		body: JSON.stringify({
			"model": llm_model,
			"messages": [
				{
					"role": "system",
					"content": prompt || "你是一个专业的AI文档助手，专注于帮助用户高效创建、编辑和优化文档内容。你的主要职责包括：1) 根据用户需求生成高质量的文档内容；2) 对现有文档进行润色、改写和优化；3) 提供文档结构建议；4) 回答与文档创作相关的专业问题。请确保你的回复专业、准确且符合文档写作规范。"
				},
				{
					"content": query,
					"role": "user"
				}
			],
			"temperature": 0.7,
			"top_p": 0.5,
			"max_tokens": 2048,
			"stream": true
		}),
	});
}